import router from '@/router'
import { useUserStore } from '@/stores/user'
import { ElMessage } from 'element-plus'
import NProgress from 'nprogress'

// NProgress配置
NProgress.configure({ showSpinner: false })

// 白名单路由
const whiteList = ['/login', '/404']

router.beforeEach(async (to, from, next) => {
  // 开始进度条
  NProgress.start()

  // 设置页面标题
  document.title = to.meta.title ? `${to.meta.title} - AI图形管理系统` : 'AI图形管理系统'

  const userStore = useUserStore()
  const token = userStore.token

  if (token) {
    // 已登录
    if (to.path === '/login') {
      // 如果已登录，访问登录页则重定向到首页
      next({ path: '/' })
      NProgress.done()
    } else {
      // 检查是否有用户信息
      if (!userStore.userInfo.id) {
        try {
          // 获取用户信息
          await userStore.getUserInfo()

          // 检查管理员权限
          if (to.meta.requiresAdmin && !userStore.isAdmin) {
            ElMessage.error('权限不足')
            next({ path: '/' })
            return
          }

          next()
        } catch (error) {
          // 获取用户信息失败，清除token并重定向到登录页
          userStore.logout()
          ElMessage.error('登录状态已过期，请重新登录')
          next(`/login?redirect=${to.path}`)
        }
      } else {
        // 检查管理员权限
        if (to.meta.requiresAdmin && !userStore.isAdmin) {
          ElMessage.error('权限不足')
          next({ path: '/' })
          return
        }

        next()
      }
    }
  } else {
    // 未登录
    if (whiteList.includes(to.path)) {
      // 在白名单中，直接进入
      next()
    } else {
      // 不在白名单中，重定向到登录页
      next(`/login?redirect=${to.path}`)
      NProgress.done()
    }
  }
})

router.afterEach(() => {
  // 结束进度条
  NProgress.done()
})